import os

def set_env():
    os.environ["TOKENIZERS_PARALLELISM"] = "false"
    os.environ["WANDB_WATCH"] = "false"
    os.environ["WANDB_LOG_MODEL"] = "false"
    os.environ["WANDB_LOG_MODEL"] = "vla-quant"
    
    
    import datasets
    datasets.config.IN_MEMORY_MAX_SIZE = 64 * 1024 * 1024 * 1024
    
def set_seed(seed):
    import numpy as np
    import torch
    import random
    
    np.random.seed(seed)
    torch.random.manual_seed(seed)
    random.seed(seed)
    
set_env()
